'''
#   测试时候使用的数据集（无用）
'''

from torchvision import datasets
from torchvision import transforms
from torch.utils.data import DataLoader

batch_size = 64
transform = transforms.Compose([transforms.ToTensor(),transforms.Normalize((0.1307,),(0.3081,))])

train_set = datasets.MNIST(root='../DB/mnist_train',
                           train=True,
                           transform=transform,
                           download=False)

train_loader = DataLoader(dataset=train_set,
                          batch_size=batch_size,
                          shuffle=True,
                          num_workers=2)

test_set = datasets.MNIST(root='../DB/mnist_test',
                          train=False,
                          transform=transform,
                          download=False)

test_loader = DataLoader(dataset=test_set,
                         batch_size=batch_size,
                         shuffle=False)